package com.linzm.leetcode.primary.exercises4_20230129;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @Author zimingl
 * @Date 2023/2/9 22:59
 * @Description: 杨辉三角
 * 给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。
 * 在「杨辉三角」中，每个数是它左上方和右上方的数的和。
 * 输入: numRows = 5
 * 输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
 * 1 <= numRows <= 30
 */
public class Demo15_118 {
    public static void main(String[] args) {
        Demo15_118 demo15_118 = new Demo15_118();
        List<List<Integer>> generate = demo15_118.generate(5);
        System.out.println(generate);
    }

    private List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < numRows; i++) {
            int[] arr = new int[i + 1];
            Arrays.fill(arr, 1);
            List<Integer> list = Arrays.stream(arr).boxed().collect(Collectors.toList());
            if (i <= 1) {
                res.add(list);
                continue;
            }
            List<Integer> tmpList = res.get(i - 1);
            for (int j = 1; j < i; j++) {
                Integer tmp = tmpList.get(j - 1) + tmpList.get(j);
                list.set(j, tmp);
            }
            res.add(list);
        }
        return res;
    }
}
